WeGR: Unified Generative Retrieval for Industrial Search Systems in WeChat
微信工业级搜索的统一生成式检索框架 WeGR
🧭 生成式检索 · SID 构建 + RL 偏好优化 ★ 深度解读
Tencent Penguin Island
联合行为感知 SID 构建与动态 horizon RL，解决生成式检索中的码本坍塌和奖励稀疏
📌 工作参考价值（4 点）
- SID 构建：将 query-item / item-item 协同信号作为对比学习损失加在连续表征上，并通过分离的 projection head 避免直接干扰 codebook；同时使用 codebook refresh（只对第一层，EMA 阈值重置）和残差保真 + 碰撞正则化，能显著提升 codebook 利用率，避免 SID 坍塌。这些 trick 可直接迁移到电商 Semantic ID 学习。
- RL 稀疏 reward：如果直接使用点击/订单作为 outcome reward，GRPO 会因 exact match 而大面积失效。借鉴 DHPO 的动态 horizon 课程，把 SID 前缀匹配作为早期 reward，逐步扩展到完整 SID，并只更新对应前缀 token，既不需要离线训练 ranker，也避免 beam search 的分布偏移。
- 多正样本 SFT：工业搜索 session 内通常有多个正反馈 item，MSFT 同时拟合所有正样本，可作为 RL 的稳定正则项，防止 policy 在稀疏 reward 下遗忘检索能力；建议与新策略 RL 联合训练，权重 0.5 左右可参考。
- 工程部署：生成式检索替换传统多路召回能直接减少进入粗排的候选量（本案 -38.10%），带来 CPU/GPU 和延迟收益；可考虑先 additive，再逐步加大替换比例验证收益。